from drf_haystack.serializers import HaystackSerializer, HaystackSerializerMixin
from haystack import indexes
from fjr.models import SKU
from fjr.serializers.sku import SKUModelSerializer


class LocationIndex(indexes.SearchIndex, indexes.Indexable):
    # SKU索引数据模型类
    text = indexes.CharField(document=True, use_template=True)
    id = indexes.IntegerField(model_attr='id')
    name = indexes.CharField(model_attr='name')
    caption = indexes.CharField(model_attr='caption')

    autocmplete = indexes.EdgeNgramField()

    @staticmethod
    def prepare_autocomplete(obj):
        return " ".join((obj.id, obj.name, obj.caption))

    def get_model(self):
        # 返回建立索引的模型类
        return SKU

    def index_queryset(self, using=None):
        # 返回要建立索引的数据库查询集
        return self.get_model().objects.filter(is_launched=True)


class LocationSerializer(HaystackSerializer):

    class Meta:
        index_classes = [LocationIndex]
        fields = ('text', 'id', 'name', 'caption')


class SKUSearchSerializer(HaystackSerializerMixin, SKUModelSerializer):

    class Meta(SKUModelSerializer.Meta):
        search_fields = ("text", )


